package com.ruoyi.common.utils.bean;

import cn.hutool.core.util.StrUtil;
import com.deepoove.poi.config.ConfigureBuilder;
import com.deepoove.poi.plugin.table.LoopRowTableRenderPolicy;
import java.io.*;
import java.lang.reflect.Field;
import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

/**
 * Bean 工具类
 */
public class BeanUtils extends org.springframework.beans.BeanUtils {
    /**
     * Bean方法名中属性名开始的下标
     */
    private static final int BEAN_METHOD_PROP_INDEX = 3;

    /**
     * 匹配getter方法的正则表达式
     */
    private static final Pattern GET_PATTERN = Pattern.compile("get(\\p{javaUpperCase}\\w*)");

    /**
     * 匹配setter方法的正则表达式
     */
    private static final Pattern SET_PATTERN = Pattern.compile("set(\\p{javaUpperCase}\\w*)");

    /**
     * Bean属性复制工具方法。
     *
     * @param dest 目标对象
     * @param src  源对象
     */
    public static void copyBeanProp(Object dest, Object src) {
        try {
            copyProperties(src, dest);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * 获取对象的setter方法。
     *
     * @param obj 对象
     * @return 对象的setter方法列表
     */
    public static List<Method> getSetterMethods(Object obj) {
        // setter方法列表
        List<Method> setterMethods = new ArrayList<Method>();

        // 获取所有方法
        Method[] methods = obj.getClass().getMethods();

        // 查找setter方法

        for (Method method : methods) {
            Matcher m = SET_PATTERN.matcher(method.getName());
            if (m.matches() && (method.getParameterTypes().length == 1)) {
                setterMethods.add(method);
            }
        }
        // 返回setter方法列表
        return setterMethods;
    }

    /**
     * 获取对象的getter方法。
     *
     * @param obj 对象
     * @return 对象的getter方法列表
     */

    public static List<Method> getGetterMethods(Object obj) {
        // getter方法列表
        List<Method> getterMethods = new ArrayList<Method>();
        // 获取所有方法
        Method[] methods = obj.getClass().getMethods();
        // 查找getter方法
        for (Method method : methods) {
            Matcher m = GET_PATTERN.matcher(method.getName());
            if (m.matches() && (method.getParameterTypes().length == 0)) {
                getterMethods.add(method);
            }
        }
        // 返回getter方法列表
        return getterMethods;
    }

    /**
     * 检查Bean方法名中的属性名是否相等。<br>
     * 如getName()和setName()属性名一样，getName()和setAge()属性名不一样。
     *
     * @param m1 方法名1
     * @param m2 方法名2
     * @return 属性名一样返回true，否则返回false
     */

    public static boolean isMethodPropEquals(String m1, String m2) {
        return m1.substring(BEAN_METHOD_PROP_INDEX).equals(m2.substring(BEAN_METHOD_PROP_INDEX));
    }

    /**
     * 复制集合
     */
    public static <T> List<T> copyList(List<T> src) throws IOException, ClassNotFoundException {
        ByteArrayOutputStream byteOut = new ByteArrayOutputStream();
        ObjectOutputStream out = new ObjectOutputStream(byteOut);
        out.writeObject(src);

        ByteArrayInputStream byteIn = new ByteArrayInputStream(byteOut.toByteArray());
        ObjectInputStream in = new ObjectInputStream(byteIn);
        @SuppressWarnings("unchecked")
        List<T> dest = (List<T>) in.readObject();
        return dest;
    }


    /**
     * 递归方法，设置LoopRowTableRenderPolicy
     */
    public static String setLoopRowTableRenderPolicy(Map<String, Object> map, String prefix, ConfigureBuilder builder) {
        LoopRowTableRenderPolicy policy = new LoopRowTableRenderPolicy();
        for (Map.Entry<String, Object> entry : map.entrySet()) {
            String currentKey = entry.getKey();
            Object value = entry.getValue();
            if (value instanceof Map) {
                // 递归调用，处理嵌套的Map
                String nestedKeys = "";
                if (StrUtil.isBlank(prefix)) {
                    nestedKeys = currentKey + ".";
                } else {
                    nestedKeys = prefix + currentKey + ".";
                }
                setLoopRowTableRenderPolicy((Map<String, Object>) value, nestedKeys, builder);
            } else if (value instanceof List) {
                String nestedKeys = "";
                // 到达Map的最底层，直接连接键
                if (StrUtil.isBlank(prefix)) {
                    nestedKeys = currentKey + ".";
                } else {
                    nestedKeys = prefix + currentKey + ".";
                }
                if (nestedKeys.length() > 0 && nestedKeys.endsWith(".")) {
                    nestedKeys = nestedKeys.substring(0, nestedKeys.length() - 1);
                }
                builder.bind(nestedKeys, policy);
            } else if (value != null && value.getClass().getPackage().getName().startsWith("com.ruoyi")) {
                String nestedKeys = "";
                if (StrUtil.isBlank(prefix)) {
                    nestedKeys = currentKey + ".";
                } else {
                    nestedKeys = prefix + currentKey + ".";
                }
                // 如果值是Java Bean，递归遍历Java Bean的字段
                traverseBean(value, nestedKeys, builder);
            }
        }

        return prefix;
    }

    private static void traverseBean(Object bean, String prefix, ConfigureBuilder builder) {
        LoopRowTableRenderPolicy policy = new LoopRowTableRenderPolicy();
        // 使用反射获取Java Bean的字段
        Field[] fields = bean.getClass().getDeclaredFields();
        for (Field field : fields) {
            try {
                field.setAccessible(true);
                Object fieldValue = field.get(bean);
                String currentKey = field.getName();
                if (fieldValue instanceof Map) {
                    // 如果字段值是Map，递归遍历
                    setLoopRowTableRenderPolicy((Map<String, Object>) fieldValue, prefix, builder);
                } else if (fieldValue instanceof List) {
                    String nestedKeys = "";
                    if (StrUtil.isBlank(prefix)) {
                        nestedKeys = currentKey + ".";
                    } else {
                        nestedKeys = prefix + currentKey + ".";
                    }
                    // 删除开头的点号
                    if (nestedKeys.length() > 0 && nestedKeys.endsWith(".")) {
                        nestedKeys = nestedKeys.substring(0, nestedKeys.length() - 1);
                    }
                    builder.bind(nestedKeys, policy);
                } else if (fieldValue != null
                        && fieldValue.getClass().getPackage().getName().startsWith("could.com.hanwen")) {
                    // 如果字段值是Java Bean，递归遍历Java Bean的字段
                    traverseBean(fieldValue, prefix, builder);
                }
            } catch (IllegalAccessException e) {
                e.printStackTrace();
            }
        }
    }


    private static void traverseBean(Object bean) {
        // 使用反射获取Java Bean的字段
        Field[] fields = bean.getClass().getDeclaredFields();
        for (Field field : fields) {
            try {
                field.setAccessible(true);
                Object fieldValue = field.get(bean);

                if (fieldValue instanceof Map) {
                    // 如果字段值是Map，递归遍历
                    formatMapNullValue((Map<String, Object>) fieldValue);
                } else if (fieldValue instanceof List) {
                    // 如果字段值是List，递归遍历
                    traverseList((List<Object>) fieldValue);
                } else if (fieldValue != null
                        && fieldValue.getClass().getPackage().getName().startsWith("could.com.hanwen")) {
                    // 如果字段值是Java Bean，递归遍历Java Bean的字段
                    traverseBean(fieldValue);
                } else if (fieldValue == null) {
                    if ("String".equals(field.getType().getSimpleName())) {
                        field.set(bean, "");
                    } else if ("Integer".equals(field.getType().getSimpleName())) {
                        field.set(bean, 0);
                    } else if ("Double".equals(field.getType().getSimpleName())) {
                        field.set(bean, 0);
                    }
                }
            } catch (IllegalAccessException e) {
                e.printStackTrace();
            }
        }
    }


    private static void traverseList(List<Object> list) {
        for (int i = 0; i < list.size(); i++) {
            Object listItem = list.get(i);
            if (listItem instanceof Map) {
                // 如果列表元素是Map，递归遍历
                formatMapNullValue((Map<String, Object>) listItem);
            } else if (listItem instanceof List) {
                // 如果列表元素是List，递归遍历
                traverseList((List<Object>) listItem);
            } else if (listItem != null && listItem.getClass().getPackage().getName().startsWith("could.com.hanwen")) {
                // 如果列表元素是Java Bean，递归遍历Java Bean的字段
                traverseBean(listItem);
            } else if (listItem == null) {
                list.set(i, "");
            }
        }
    }

    /**
     * 遍历map，对基本类型null 初始化
     */
    public static void formatMapNullValue(Map<String, Object> map) {
        for (Map.Entry<String, Object> entry : map.entrySet()) {
            String key = entry.getKey();
            Object value = entry.getValue();

            if (value instanceof Map) {
                // 如果值是Map，递归遍历
                formatMapNullValue((Map<String, Object>) value);
            } else if (value instanceof List) {
                // 如果值是List，遍历List中的元素
                traverseList((List<Object>) value);
            } else if (value != null && value.getClass().getPackage().getName().startsWith("could.com.hanwen")) {
                // 如果值是Java Bean，递归遍历Java Bean的字段
                traverseBean(value);
            } else {
                if (value == null) {
                    map.put(key, "");
                }
            }
        }
    }

}
